import os
import os.path

#=============================================================================    
#自定义函数：遍历目录，计算文件大小

list1=[]                               #定义一个空的列表，用来累计每个文件的大小
def countdirsize(dir):    
    for f in os.listdir(dir):          #遍历目录下所有文件
        filepath=os.path.join(dir,f)
        if os.path.isfile(filepath):   #判断是否为文件
            list1.append(os.path.getsize(filepath))  #计算文件大小并累加
            
        if os.path.isdir(filepath):    #判断是否为目录
            countdirsize(filepath)     #如果是目录，则递归
                    
#=============================================================================

countdirsize('./test2')                #调用自定义函数，计算文件大小
print(sum(list1))                      #输出文件大小值